# -*- coding:utf-8 -*-
"""
Author：Administrator
Date:2021年10月14日
"""
import pandas as pd

path = './数据/附加阻尼比计算原始数据.xlsx'

data_sf = pd.read_excel(path, sheet_name='楼层剪力', skiprows=2, usecols=[0, 2, 3, 4], names=['层', '时程', 'X向剪力(kN)',
                                                                                          'Y向剪力(kN)'])

# 去除不包含屋顶行的数据
data_sf = data_sf[~data_sf['层'].str.contains('屋顶')]
# 筛选只包含'全部'的数据行
con = data_sf['时程'].str.contains('全部')
data_sf = data_sf[con].reset_index(drop=True)

data_sf.loc[data_sf['时程'].str.contains('X'), '楼层剪力(kN)'] = data_sf['X向剪力(kN)']
data_sf.loc[data_sf['时程'].str.contains('Y'), '楼层剪力(kN)'] = data_sf['Y向剪力(kN)']

data_sf['楼层剪力(kN)'] = abs(data_sf['楼层剪力(kN)'])
print(data_sf)

data_sf.to_excel('./数据/01结果-剪力.xlsx', index=False)
